working with functions in python